git.r.bdr.sh - rbdr/captura/blob - Captura/Core Extensions/CVImageBuffer+cgImage.swift

   1 /*
   2  Copyright (C) 2024 Rubén Beltrán del Río
   3
   4  This program is free software: you can redistribute it and/or modify
   5  it under the terms of the GNU General Public License as published by
   6  the Free Software Foundation, either version 3 of the License, or
   7  (at your option) any later version.
   8
   9  This program is distributed in the hope that it will be useful,
  10  but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12  GNU General Public License for more details.
  13
  14  You should have received a copy of the GNU General Public License
  15  along with this program. If not, see https://captura.tranquil.systems.
  16  */
  17 import Foundation
  18 import ReplayKit
  19
  20 extension CVImageBuffer {
  21
  22   private static let contextQueue = DispatchQueue(label: "com.example.contextQueue")
  23   static let sharedContext: CIContext = {
  24     return CIContext()
  25   }()
  26
  27   var cgImage: CGImage? {
  28     var result: CGImage?
  29     CVImageBuffer.contextQueue.sync {
  30       let ciImage = CIImage(cvImageBuffer: self)
  31       let width = CVPixelBufferGetWidth(self)
  32       let height = CVPixelBufferGetHeight(self)
  33       result = CVImageBuffer.sharedContext.createCGImage(
  34         ciImage, from: CGRect(x: 0, y: 0, width: width, height: height))
  35     }
  36     return result
  37   }
  38 }